import torch
from diffusers import StableDiffusionPipeline

# pipe = StableDiffusionPipeline.from_pretrained(
#     "emilianJR/chilloutmix_NiPrunedFp32Fix"
#     , torch_dtype = torch.float16
#     , safety_checker = None
#     , cache_dir="diffusers-cache"
# ).to("cuda")
#
# lora_path = "lora/1.5/blindbox_v1_mix.safetensors"
# pipe.load_lora_weights(lora_path)


# pipe = StableDiffusionPipeline.from_pretrained(
#     "stabilityai/stable-diffusion-2-1-base"
#     , torch_dtype = torch.float16
#     , safety_checker = None
# ).to("cuda")
#
# lora_path = "lora/2.1/hjjh.safetensors"
# pipe.load_lora_weights(lora_path)


pipe = StableDiffusionPipeline.from_pretrained(
    "runwayml/stable-diffusion-v1-5"
    , torch_dtype = torch.float16
    , safety_checker = None
).to("cuda")

lora_path = "lora/1.5/blindbox_v1_mix.safetensors"
pipe.load_lora_weights(lora_path)